#include <iostream>
#include <string>
#include <vector>

// 联系人
struct Contact
{
    std::string name;
    int age;
    int gender;
    std::string phone_number;
    std::string address;
};

std::vector<Contact> g_contact_lst;

void help()
{
    std::vector<std::string> message_list = {
        "1、添加联系人",
        "2、显示联系人",
        "3、删除联系人",
        "4、查找联系人",
        "5、修改联系人",
        "6、清空联系人",
        "0、退出通讯录"
    };

    std::cout << std::string(29, '*') << std::endl;
    for (int i = 0; i < message_list.size(); i++)
    {
        std::cout << std::string(5, '*') << "   " << message_list[i] << "   " << std::string(5, '*') << std::endl;
    }
    std::cout << std::string(29, '*') << std::endl;
}

void clear_screen()
{
    std::cout << "\033[2J\033[1;1H"; // clear screen
}

void pause()
{
    std::cout << "按任意键继续..." << std::endl;
    getchar();
}

Contact input_contact()
{
    Contact contact;

    std::cout << "请输入姓名: ";
    std::cin >> contact.name; // TODO: 有空格怎么处理？

    std::cout << "请输入性别:" << std::endl;
    std::cout << "1 -- 男" << std::endl;
    std::cout << "2 -- 女" << std::endl;
    std::cin >> contact.gender;

    std::cout << "请输入年龄: ";
    std::cin >> contact.age;

    std::cout << "请输入联系电话: ";
    std::cin >> contact.phone_number;

    std::cout << "请输入家庭住址: ";
    std::cin >> contact.address;

    return contact;
}

// 添加联系人
void add_contact()
{
    Contact contact = input_contact();
    g_contact_lst.push_back(contact);
    std::cout << "添加成功" << std::endl;
}

void show_one_contact(Contact& contact)
{
    std::string gender_str[] = {
        "",
        "男",
        "女"
    };
    std::cout << "姓名：" << contact.name << "\t";
    std::cout << "性别：" << gender_str[contact.gender] << "\t";
    std::cout << "年龄：" << contact.age << "\t";
    std::cout << "电话：" << contact.phone_number << "\t";
    std::cout << "住址：" << contact.address << "\t";
    std::cout << std::endl;
}

// 显示联系人
void show_contact()
{
    if (g_contact_lst.size() > 0)
    {
        for (int i = 0; i < g_contact_lst.size(); i++)
        {
            show_one_contact(g_contact_lst[i]);
        }
    }
    else
    {
        std::cout << "当前记录为空" << std::endl;
    }
}

int find_contact_by_name(std::string name)
{
    int index = -1;
    for (int i = 0; i < g_contact_lst.size(); i++)
    {
        Contact& contact = g_contact_lst[i];
        if (contact.name == name)
        {
            index = i;
            break;
            // 这里的问题：只删除了第一次出现的联系人
            // 假设有同名联系人，如果全都要删掉，那么erase方法被多次调用，但index offset也在不断变化
            // 这就需要把g_contact_lst从vector类型，改为链表类型
        }
    }
    return index;
}

// 删除联系人
void delete_contact()
{
    std::string name;
    std::cout << "请输入您要删除的联系人" << std::endl;
    std::cin >> name;
    int index = find_contact_by_name(name);

    if (index >= 0)
    {
        g_contact_lst.erase(g_contact_lst.begin() + index);
        std::cout << "删除成功" << std::endl;
    }
    else
    {
        std::cout << "查无此人" << std::endl;
    }
}

// 查找联系人
void search_contact()
{
    std::string name;
    std::cout << "请输入您要查找的联系人" << std::endl;
    std::cin >> name;
    int index = find_contact_by_name(name);
    if (index >= 0)
    {
        show_one_contact(g_contact_lst[index]);
    }
    else
    {
        std::cout << "查无此人" << std::endl;
    }
}

// 修改联系人
void modify_contact()
{
    std::string name;
    std::cout << "请输入您要修改的联系人" << std::endl;
    std::cin >> name;
    int index = find_contact_by_name(name);
    if (index >= 0)
    {
        Contact contact = input_contact();
        g_contact_lst[index] = contact;
        std::cout << "修改成功" << std::endl;
    }
    else
    {
        std::cout << "查无此人" << std::endl;
    }
}

// 清空联系人
void clear_all_contact()
{
    g_contact_lst.clear();
}

// 退出通讯录
void exit_system()
{
    std::cout << "欢迎下次使用" << std::endl;
    getchar(); // 吸收掉换行符
    pause();
}

void show_error_message()
{
    std::cout << "选择非法！请重新选择" << std::endl;
}

int main()
{
    int option;
    while (true)
    {
        clear_screen();
        help();
        std::cin >> option;
        switch (option)
        {
        case 1:
            add_contact();
            break;
        case 2:
            show_contact();
            break;
        case 3:
            delete_contact();
            break;
        case 4:
            search_contact();
            break;
        case 5:
            modify_contact();
            break;
        case 6:
            clear_all_contact();
            break;
        case 0:
            exit_system();
            return 0;
        default:
            show_error_message();
        }
        getchar(); // 吸收掉换行符
        pause();
    }

    return 0;
}